from os import listdir
#这里要用到函数表达
#输入关键字，指定文件夹中的所有文件名中含有此关键字的文件/目录
def doc(path,key):
    result=listdir(path)
    if key in result:
        print(key)
    return result
#文件内容中含有此关键字的文件。
    for i in result:
        with open(i) as f:
            cont=f.readline()
            if key in cont:
                print(i)
path=input('print your path:')
key=input('print your key word:')
doc(path,key)